package com.uairobot.bill.jfinal.generator;

import javax.sql.DataSource;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;
import com.uairobot.bill.jfinal.utils.PropKitUtil;

public class JFinalGenerator {

  public static String configFileName = PropKitUtil.configFileName;
  public static Prop p;

  private static void loadConfig() {
    if (p == null) {
      p = PropKit.use(configFileName);
    }
  }

  public static DruidPlugin createDruidPlugin() {
    loadConfig();
    return new DruidPlugin(p.get("jdbc.url"), p.get("jdbc.user"), p.get("jdbc.pswd").trim());
  }

  public static DataSource getDataSource() {
    DruidPlugin druidPlugin = createDruidPlugin();
    druidPlugin.start();
    return druidPlugin.getDataSource();
  }

  public static void main(String[] args) {
    String packageName = "com.uairobot.bill.freeswitch.model";
    // base model 所使用的包名
    String baseModelPackageName = packageName + ".base";
    // base model 文件保存路径
    String replace = baseModelPackageName.replace(".", "/");
    String baseModelOutputDir = PathKit.getWebRootPath() + "/src/main/java/" + replace;

    // model 所使用的包名 (MappingKit 默认使用的包名)
    String modelPackageName = packageName;
    // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
    String modelOutputDir = baseModelOutputDir + "/..";

    // 创建生成器
    Generator generator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName,
        modelOutputDir);

    // 配置是否生成备注
    generator.setGenerateRemarks(true);

    // 设置数据库方言
    generator.setDialect(new MysqlDialect());

    // 设置是否生成链式 setter 方法
    generator.setGenerateChainSetter(false);

    // 添加不需要生成的表名
    generator.addExcludedTable("adv");

    // 设置是否在 Model 中生成 dao 对象
    generator.setGenerateDaoInModel(false);

    // 设置是否生成字典文件
    generator.setGenerateDataDictionary(false);

    // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非
    // OscUser
    generator.setRemovedTableNamePrefixes("t_");

    // 生成
    generator.generate();
  }
}
